package leetcode.每日一题;

import org.junit.Test;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

/**
 * @author ：zsy
 * @date ：Created 2022/11/14 9:34 PM
 * @description：
 */
public class 数组的均值分割 {
    @Test
    public void test() {

    }

    class Solution {
        public boolean splitArraySameAverage(int[] nums) {
            int n = nums.length;
            if (n == 1) return false;
            int sum = Arrays.stream(nums).sum();
            // 处理原数组，防止出现负数
            for (int i = 0; i < n; i++) {
                nums[i] = nums[i] * n - sum;
            }
            int m = n / 2;
            // 在左边找Ao
            Set<Integer> left = new HashSet<>();
            for (int i = 1; i < (1 << m); i++) {
                int tot = 0;
                for (int j = 0; j < m; j++) {
                    if ((i & (1 << j))!= 0) {
                        tot += nums[j];
                    }
                }
                if (tot == 0) {
                    return true;
                }
                left.add(tot);
            }
            int rsum = 0;
            for (int i = m; i < n; i++) {
                rsum += nums[i];
            }
            // 在右边找
            for (int i = 1; i < (1 << (n - m)); i++) {
                int tot = 0;
                for (int j = m; j < n; j++) {
                    if ((i & (1 << (j - m))) != 0) {
                        tot += nums[j];
                    }
                }
                if (tot == 0 || (tot != rsum && left.contains(-tot))) {
                    return true;
                }
            }
            return false;
        }
    }
}